
#coding=utf-8
 
from mmdet.apis import init_detector
from mmdet.apis import inference_detector
from mmdet.apis import show_result_pyplot
 
# 模型配置文件
config_file = '/DATA/ybli/mmdetection-openvino/configs/0mine/KT-0105.py'
# 预训练模型文件
checkpoint_file = '/DATA/ybli/mmdetection-openvino/models/epoch_20/ckpt.pth'
 
# 通过模型配置文件与预训练文件构建模型
model = init_detector(config_file, checkpoint_file, device='cuda:0')
 
# 测试单张图片并进行展示
img = '/DATA/ybli/D_datas/JPEGImages/3_2.jpg'
result = inference_detector(model, img)
model.show_result(img, result, score_thr=0.3, show=False, out_file='out.jpg')